#!/bin/bash
#SBATCH -J task_base
#SBATCH -p gpu-normal
#SBATCH -N 1
#SBATCH --gres=gpu:1
#SBATCH -n 1
#SBATCH -o task_base.out
#SBATCH -e task_base.err


module load anaconda3/2019.10 cuda/10.1

cd ..
pwd

python train.py --gpu_id=0 --output_dir=output_base/gan1 --GAN_type=gan --is_load=True --iteration=5000 --log_interval=100 --gru_type=grui --num_layer=2 --hidden_dim=64  --seq_len=12 --impute_iter=2000
python train.py --gpu_id=0 --output_dir=output_base/gan2 --manualseed=1028 --GAN_type=gan --is_load=True --iteration=5000 --log_interval=100 --gru_type=grui --num_layer=2 --hidden_dim=64  --seq_len=12 --impute_iter=2000
python train.py --gpu_id=0 --output_dir=output_base/wgan --GAN_type=WGAN --is_load=True --iteration=5000 --log_interval=100 --gru_type=grui --num_layer=2 --hidden_dim=64  --seq_len=12 --impute_iter=2000

